package com.jim.healio.agents.impl;

import com.jim.healio.agents.Agent;
import com.jim.healio.tools.GraphRAGTool;
import com.jim.healio.utils.PromptUtil;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.stereotype.Component;

@Component
public class Doctor extends Agent {

    public Doctor(ChatClient.Builder chatClientBuilder) {
        super(
                "主治医师",
                "Bob",
                chatClientBuilder
                        .defaultSystem(PromptUtil.getDoctorSystemPrompt())
                        .defaultTools(new GraphRAGTool())
                        .build());
    }

}
